# -*- coding:utf-8 -*-
# @Time: 2020/9/19 10:26
# @Author: 周铖鑫
# @Description：contours等高线图
# @File: UseContours.py
import  matplotlib.pyplot as plt
import numpy as np
def f(x,y):
    #the height function
    return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
n=256
x=np.linspace(-3,3,n)
y=np.linspace(-3,3,n)
X,Y=np.meshgrid(x,y)
#8表示填充8种颜色
plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)
#8表示8条等高线
C=plt.contour(X,Y,f(X,Y),8,colors='black',linewidths=.5)
#add label
plt.clabel(C,inline=True,fontsize=10)
plt.xticks(())
plt.yticks(())
plt.show()